package main
 
import (
        "fmt"
        "io"
        "net/http"
        "net/url"
        "os"
        "os/exec"
        "path/filepath"
        "strconv"
        "strings"
)
 
// 定义一个全局变量name，用于存储文件名
var name = ""
 
// WriteCounter是一个用于统计写入字节数的结构体
type WriteCounter struct {
        Total uint64
}
 
// Write方法实现了io.Writer接口，用于统计写入的字节数并打印进度
func (wc *WriteCounter) Write(p []byte) (int, error) {
        n := len(p)
        wc.Total += uint64(n)
        wc.PrintProgress()
        return n, nil
}
 
// PrintProgress方法用于打印下载进度
func (wc WriteCounter) PrintProgress() {
        // 清除之前的进度信息
        fmt.Printf("\r%s", strings.Repeat(" ", 35))
        // 打印新的进度信息，单位为MB
        fmt.Printf("\r文件已下载： %d MB ", wc.Total/(1024*1024))
}
 
func main() {
        // 检查命令行参数数量，如果不足2个则提示错误并返回
        if len(os.Args) < 2 {
                fmt.Println("请提供至少一个参数！")
                return
        }
 
        // 获取命令行参数
        arg := os.Args[1]
        var aru = os.Args[2]
 
        // 打印下载地址和主程序进程信息
        fmt.Println("下载地址:", arg)
        fmt.Println("主程序进程:", aru)
 
        // 杀死指定名称的进程
        exec.Command(`taskkill`, `/f`, `/t`, `/im`, aru).Run()
 
        // 从URL中提取文件名和文件大小
        fileName, err := extractFileNameFromURL(arg)
        if err != nil {
                fmt.Println("Error:", err)
                return
        }
        fileSize, err := getFileSizeFromURL(arg)
        if err != nil {
                fmt.Println("Error:", err)
                return
        }
 
        // 打印正在下载的提示信息和文件大小
        fmt.Println("正在下载 请稍后.......")
        fmt.Printf("文件大小:   %.2f MB\n", fileSize)
 
        // 开始下载文件
        name = fileName
        fileUrl := arg
        err = DownloadFile(fileName, fileUrl)
        if err != nil {
                panic(err)
        }
 
        // 下载完成打印提示信息
        fmt.Println("下载完成")
}
 
// DownloadFile函数用于下载文件
func DownloadFile(filepath string, url string) error {
        // 创建文件
        out, err := os.Create(filepath + ".tmp")
        if err != nil {
                return err
        }
        defer out.Close()
 
        // 发送HTTP GET请求
        resp, err := http.Get(url)
        if err != nil {
                return err
        }
        defer resp.Body.Close()
 
        // 使用WriteCounter统计写入字节数
        counter := &WriteCounter{}
        _, err = io.Copy(out, io.TeeReader(resp.Body, counter))
        if err != nil {
                return err
        }
 
        // 关闭文件
        fmt.Print("\n")
        out.Close()
 
        // 重命名文件
        err = os.Rename(filepath+".tmp", filepath)
        if err != nil {
                return err
        }
 
        // 执行下载的文件
        exec.Command(".\\" + name).Run()
        os.Exit(0)
        return nil
}
 
// extractFileNameFromURL函数用于从URL中提取文件名
func extractFileNameFromURL(downloadURL string) (string, error) {
        // 解析URL
        parsedURL, err := url.Parse(downloadURL)
        if err != nil {
                return "", err
        }
 
        // 获取路径部分
        path := parsedURL.Path
 
        // 提取文件名
        fileName := filepath.Base(path)
        return fileName, nil
}
 
// getFileSizeFromURL函数用于从URL中获取文件大小
func getFileSizeFromURL(downloadURL string) (float64, error) {
        // 发送HEAD请求
        resp, err := http.Head(downloadURL)
        if err != nil {
                return 0, err
        }
        defer resp.Body.Close()
 
        // 从响应头中获取Content-Length字段
        contentLengthStr := resp.Header.Get("Content-Length")
        if contentLengthStr == "" {
                return 0, fmt.Errorf("Content-Length header not found")
        }
 
        // 将Content-Length字符串转换为整数
        contentLength, err := strconv.ParseInt(contentLengthStr, 10, 64)
        if err != nil {
                return 0, err
        }
 
        // 将字节转换为MB
        return float64(contentLength) / (1024 * 1024), nil
}